package main

import (
	"fmt"
	"io"
	"net"
	"os"
)

func main() {
	//提示输入文件
	fmt.Println("请输入需要传输的文件：")

	var path string
	fmt.Scan(&path)

	// 获取文件名
	info, err := os.Stat(path)
	if err != nil {
		fmt.Println("os.Stat err： ", err)
		return
	}

	info.Name()

	// 主动连接服务器
	conn, err := net.Dial("tcp", "127.0.0.1:8080")
	if err != nil {
		fmt.Println("net.Dial err： ", err)
		return
	}

	defer conn.Close()

	//给接收方，先发送文件名
	_, err = conn.Write([]byte(info.Name()))
	if err != nil {
		fmt.Println("conn.Writ err： ", err)
		return
	}

	// 接收对方的回复，如果回复ok， 说明对方准备好，可以发文件
	buf := make([]byte, 1024)

	read, err := conn.Read(buf)
	if err != nil {
		fmt.Println("conn.Read err： ", err)
		return
	}

	if "ok" == string(buf[:read]) {
		SendFile(path, conn)
	}

}

func SendFile(path string, conn net.Conn) {
	// 以只读方式打开文件
	open, err := os.Open(path)
	if err != nil {
		fmt.Println("os.Open err ", err)
		return
	}
	defer open.Close()

	// 读文件内容 读多少，就给接收方发多少，一点不差
	buf := make([]byte, 1024*4) //4k

	for {
		n, err := open.Read(buf)
		if err != nil {
			if err == io.EOF {
				fmt.Println("文件发送完毕")
			} else {
				fmt.Println("open. Read err ", err)
			}
			return
		}

		// 发送文件
		conn.Write(buf[:n])
	}
}
